package timer;

import java.util.PriorityQueue;

//通过这个类来描述一个任务 注意:需要自定义比较规则
class MyTimerTask implements Comparable<MyTimerTask> {
    //在什么时间点来执行这个任务
    //此处约定这个time是一个ms级别的时间戳
    private long time;
    private Runnable runnable;

    public long getTime() {
        return time;
    }

    //此处delay是一个(相对时间)
    public MyTimerTask(long delay, Runnable runnable) {
        //计算绝对时间
        this.time = System.currentTimeMillis() + delay;
        this.runnable = runnable;
    }

    public void run() {
        runnable.run();
    }

    @Override
    public int compareTo(MyTimerTask o) {
        return (int) (this.time - o.time);//此处需要强转
    }
}

//通过这个类来表示一个定时器
class MyTimer {
    //负责扫描任务队列 执行任务的线程
    private Thread t = null;
    //任务队列
    private PriorityQueue<MyTimerTask> queue = new PriorityQueue<>();
    private Object locker = new Object();

    public void schedule(Runnable runnable, long delay) {
        synchronized (locker) {
            MyTimerTask task = new MyTimerTask(delay, runnable);
            queue.offer(task);
            locker.notify();//入队列后唤醒
        }
    }

    //构造方法 创建扫描线程 让扫描线程来完成判定和执行
    public MyTimer() { //默认前台线程 阻止线程结束
        t = new Thread(() -> {
            //扫描线程就需要循环往复地扫描队首元素,然后判定队首元素是不是时间到了
            //如果时间没到,不进行任何操作
            //如果时间到了,就执行该任务并把这个任务从队列中删除
            while (true) {
                try {
                    synchronized (locker) { //若加在while外无法释放锁
                        while (queue.isEmpty()) {
                            // 暂时先不处理
                            locker.wait();
                        }
                        MyTimerTask task = queue.peek();
                        //获取到当前时间
                        long curTime = System.currentTimeMillis();
                        if (curTime >= task.getTime()) {
                            //当前时间已经到任务时间,可以执行任务
                            queue.poll();
                            task.run();
                        } else {
                            //当前时间还没到 暂时先不执行
                            //Thread.sleep(task.getTime() - curTime); //不建议使用sleep 若该时间段内有新任务无法解锁执行
                            locker.wait(task.getTime() - curTime);
                        }
                    } //释放锁
                } catch (InterruptedException e) {
                    throw new RuntimeException(e);
                }
            }
        });
        t.start();
    }
}

public class Implement_Timer {
    public static void main(String[] args) {
        //在主线程中对队列元素进行添加
        MyTimer timer = new MyTimer();
        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello 3000");
            }
        }, 3000);
        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello 2000");
            }
        }, 2000);
        timer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("hello 1000");
            }
        }, 1000);
        System.out.println("hello main");
    }
}
